import os
import json
from tqdm import tqdm
from tools.combine_utils import combine_texts_to_mp3

def process_json_file(json_file_path, is_chinese1, is_chinese2):
    # 获取JSON文件的名称和目录
    file_name = os.path.basename(json_file_path)  # 例如: "example.json"
    file_dir = os.path.dirname(json_file_path)  # 获取文件所在目录
    folder_name = os.path.splitext(file_name)[0]  # 去掉文件扩展名, 例如 "example"

    # 在同目录下创建与json文件名相同的文件夹
    output_folder = os.path.join(file_dir, folder_name)
    if not os.path.exists(output_folder):
        os.makedirs(output_folder)
        print(f"创建了文件夹: {output_folder}")

    # 读取 JSON 文件
    with open(json_file_path, 'r', encoding='utf-8') as json_file:
        data = json.load(json_file)

    # 遍历 JSON 中的 pairs 数组
    for pair in tqdm(data['pairs'], desc="爬取进度"):
        # 去掉 pair1 和 pair2 中的换行符
        pair1 = pair['pair1'].replace('\n', '')
        pair2 = pair['pair2'].replace('\n', '')
        pair_id = pair['id']

        # 设置输出文件名为 {id}.mp3
        output_file = os.path.join(output_folder, f"{pair_id}.mp3")

        # 调用 combine_texts_to_mp3 方法生成 MP3 文件
        combine_texts_to_mp3(pair1, is_chinese1, pair2, is_chinese2, output_file, pause_duration_ms=200)

# 示例调用
json_file_path = '/Users/duyi/UnityProject/draw_turtle_resource/data/english_chinese.json'  # 输入你的 JSON 文件路径
process_json_file(json_file_path, is_chinese1=False, is_chinese2=True)